/*
 * Copyright (c) 2011-2023 Contributors to the Eclipse Foundation
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License 2.0 which is available at
 * http://www.eclipse.org/legal/epl-2.0, or the Apache License, Version 2.0
 * which is available at https://www.apache.org/licenses/LICENSE-2.0.
 *
 * SPDX-License-Identifier: EPL-2.0 OR Apache-2.0
 */
package io.vertx.oracleclient.impl.commands;

import io.vertx.core.Future;
import io.vertx.core.internal.ContextInternal;
import io.vertx.core.json.JsonArray;
import io.vertx.oracleclient.OraclePrepareOptions;
import io.vertx.sqlclient.internal.PreparedStatement;
import io.vertx.sqlclient.spi.protocol.PrepareStatementCommand;
import oracle.jdbc.OracleConnection;

import java.sql.SQLException;
import java.sql.Statement;

public class OraclePrepareStatementCommand extends OracleCommand<PreparedStatement> {

  private final OraclePrepareOptions options;
  private final String sql;

  public OraclePrepareStatementCommand(OracleConnection oracleConnection, ContextInternal connectionContext, PrepareStatementCommand cmd) {
    super(oracleConnection, connectionContext);
    this.options = OraclePrepareOptions.createFrom(cmd.options());
    this.sql = cmd.sql();
  }

  @Override
  protected Future<PreparedStatement> execute() {
    boolean autoGeneratedKeys = options == null || options.isAutoGeneratedKeys();
    boolean autoGeneratedIndexes = options != null && options.getAutoGeneratedKeysIndexes() != null;

    if (autoGeneratedKeys && !autoGeneratedIndexes) {
      return prepareReturningKey();
    } else if (autoGeneratedIndexes) {
      return prepareWithAutoGeneratedIndexes();
    } else {
      return prepare();
    }
  }

  private Future<PreparedStatement> prepareWithAutoGeneratedIndexes() {
    return executeBlocking(() -> {
      // convert json array to int or string array
      JsonArray indexes = options.getAutoGeneratedKeysIndexes();
      if (indexes.getValue(0) instanceof Number) {
        int[] keys = new int[indexes.size()];
        for (int i = 0; i < keys.length; i++) {
          keys[i] = indexes.getInteger(i);
        }
        try (java.sql.PreparedStatement statement = oracleConnection.prepareStatement(sql, keys)) {
          return new OraclePreparedStatement(sql, statement);
        }
      }
      if (indexes.getValue(0) instanceof String) {
        String[] keys = new String[indexes.size()];
        for (int i = 0; i < keys.length; i++) {
          keys[i] = indexes.getString(i);
        }
        try (java.sql.PreparedStatement statement = oracleConnection.prepareStatement(sql, keys)) {
          return new OraclePreparedStatement(sql, statement);
        }
      }
      throw new SQLException("Invalid type of index, only [int, String] allowed");
    });
  }

  private Future<PreparedStatement> prepareReturningKey() {
    return executeBlocking(() -> {
      try (java.sql.PreparedStatement statement = oracleConnection.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS)) {
        return new OraclePreparedStatement(sql, statement);
      }
    });
  }

  private Future<PreparedStatement> prepare() {
    return executeBlocking(() -> {
      try (java.sql.PreparedStatement statement = oracleConnection.prepareStatement(sql)) {
        return new OraclePreparedStatement(sql, statement);
      }
    });
  }
}
